export dataset_name=/home/user4/workspace/ddg-benchmark/data

# RITA_s RITA_m RITA_l RITA_xl
# example for dataset evaluation
CUDA_VISIBLE_DEVICES=5 python src/RITA/predict.py \
--dataset_input $dataset_name \
--model_location lightonai/RITA_s lightonai/RITA_m lightonai/RITA_l lightonai/RITA_xl